#!/bin/bash
#SBATCH -J infAA100
#SBATCH -N 1
#SBATCH -p a01
#SBATCH -x g02
#SBATCH -o stdout.%j.txt
#SBATCH -e stderr.%j.txt
#SBATCH --no-requeue #作业失败后不会重新排序
#SBATCH --cpus-per-task=8 # 每个任务分配8个CPU核心（对应num_workers）
#SBATCH --ntasks-per-node=1 # 每个节点8个任务(对应8个GPU)，它只是告诉 SLURM 你想要 8 个任务（任务 ≠ 进程），而非自动运行你的 Python 脚本 8 次
#SBATCH --gres=gpu:1 #根据需要设置，只使用CPU此行省略
source ~/.bashrc
source /home/fit/liuyebin/WORK/miniconda3/bin/activate
conda activate cogvideo2

export TOKENIZERS_PARALLELISM=false
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
export CUDA_LAUNCH_BLOCKING=1

# python sanqi_test/pose2vid.py
python sanqi_test/pose2vid_train100.py